package com.andnnl.map;

import java.util.HashMap;

/**
 * Created by chenss on 2021/10/8.
 */
public class MapComputeTest {
    public static void main(String[] args) {
        computeTest();
        System.out.println("--------------");
        computeIfAbsentTest();
        System.out.println("--------------");
        computeIfPresentTest();

    }

    /**
     * computeIfPresent 的方法,对 指定的 在map中已经存在的key的value进行操作。只对已经存在key的进行操作，其他不操作
     */
    private static void computeIfPresentTest() {
        HashMap<String,Integer> map = new HashMap<>();
        map.put("1",1);
        map.put("2",2);
        map.put("3",3);
        //只对map中存在的key对应的value进行操作
        Integer integer = map.computeIfPresent("3", (k,v) -> v+1 );
        Integer integer1 = map.computeIfPresent("4", (k,v) -> {
            if (v==null)return 0;
            return v+1;
        } );
        System.out.println(integer);
        System.out.println(integer1);
        System.out.println(map.toString());
    }

    /**
     * omputeIfAbsent的方法有两个参数 第一个是所选map的key，第二个是需要做的操作。这个方法当key值不存在时才起作用。
     *
     * 当key存在返回当前value值，不存在执行函数并保存到map中
     */
    private static void computeIfAbsentTest() {
        HashMap<String,Integer> map = new HashMap<>();
        map.put("1",1);
        map.put("2",2);
        map.put("3",3);
        Integer integer = map.computeIfAbsent("3", key -> new Integer(4));//key存在返回value
        Integer integer1 = map.computeIfAbsent("4", key -> new Integer(4));//key不存在执行函数存入
        System.out.println(integer);
        System.out.println(integer1);
        System.out.println(map.toString());
    }

    /**
     * compute的方法,指定的key在map中的值进行操作 不管存不存在，操作完成后保存到map中
     */
    private static void computeTest() {
        HashMap<String,Integer> map = new HashMap<>();
        map.put("1",1);
        map.put("2",2);
        map.put("3",3);
        Integer integer = map.compute("3", (k,v) -> v+1 );
        //key不管存在不在都会执行后面的函数，并保存到map中
        Integer integer1 = map.compute("4", (k,v) -> {
            if (v==null)return 0;
            return v+1;
        } );
        System.out.println(integer);
        System.out.println(integer1);
        System.out.println(map.toString());
    }
}
